import pandas as pd

def convert_csv_to_json(csv_file_path, json_file_path, orient='records'):
    """
    将CSV文件转换为JSON格式并保存
    
    参数:
    csv_file_path (str): CSV文件路径
    json_file_path (str): JSON文件保存路径
    orient (str): JSON的输出格式，默认为'records'，即列表形式
    
    返回:
    bool: 成功返回True，失败返回False
    """
    try:
        # 读取CSV文件
        df = pd.read_csv(csv_file_path)
        
        # 根据列数重命名列名
        if len(df.columns) == 2:
            df.columns = ['ip_type_4_number', 'unifiedSocialCreditCode']
        elif len(df.columns) == 4:
            df.columns = ['ip_type_1_number', 'ip_type_2_number', 'ip_type_3_number', 'unifiedSocialCreditCode']
        
        # 将数据保存为JSON格式
        df.to_json(json_file_path, orient=orient, force_ascii=False, indent=2)
        
        print(f"成功将CSV文件 '{csv_file_path}' 转换为JSON格式并保存至 '{json_file_path}'")
        return True
    except FileNotFoundError:
        print(f"错误: 找不到CSV文件 '{csv_file_path}'")
        return False
    except Exception as e:
        print(f"错误: 转换过程中出现异常: {e}")
        return False

if __name__ == "__main__":
    # 示例使用
    csv_file = './output1.csv'     # 替换为你的CSV文件路径
    json_file = './patent.json'  # 替换为你想保存的JSON文件路径
    
    # 执行转换
    convert_csv_to_json(csv_file, json_file)    